model_id_or_path=$1
data_path=$2
output_path=$3

# model_id=Qwen/Qwen2.5-3B-Instruct
model_id=Qwen/Qwen3-1.7B

if [ ! -z "$model_id_or_path" ];then
    swift sft \
        --model $model_id \
        --model_id_or_path $output_path/${model_id_or_path} \
        --dataset $data_path \
        --num_train_epochs 5 \
        --train_type lora \
        --output_dir $output_path \
        --eval_steps 200
fi

if [ -z "$model_id_or_path" ];then
    swift sft \
        --model $model_id \
        --dataset $data_path \
        --num_train_epochs 10 \
        --train_type lora \
        --output_dir $output_path \
        --eval_steps 200
fi
